using System.Collections.Generic;

namespace Test.ConsoleProgram.Algorithm.Solution
{
    [TestDescription("算法: 0139. 单词拆分")]
    public class No0139_WordBreak : AbsBaseTestItem
    {
        /*
        给定一个非空字符串 s 和一个包含非空单词列表的字典 wordDict，判定 s 是否可以被空格拆分为一个或多个在字典中出现的单词。

        说明：

            拆分时可以重复使用字典中的单词。
            你可以假设字典中没有重复的单词。
        */

        public override void OnTest()
        {
            Assert.TestExe(WordBreak, "leetcode", new string[] { "leet", "code" }, true);
            Assert.TestExe(WordBreak, "applepenapple", new string[] { "apple", "pen" }, true);
            Assert.TestExe(WordBreak, "catsandog", new string[] { "cats", "dog", "sand", "and", "cat" }, false);
        }

        public bool WordBreak(string s, IList<string> wordDict)
        {
            var wordDictSet = new HashSet<string>(wordDict);
            var n = s.Length;
            var dp = new bool[n + 1];
            dp[0] = true;
            for (int i = 1; i <= n; i++)
            {
                for (int j = 0; j < i; j++)
                {
                    if (dp[j] && wordDictSet.Contains(s.Substring(j, i - j)))
                    {
                        dp[i] = true;
                        break;
                    }
                }
            }
            return dp[s.Length];
        }
    }
}
